import tensorrt as trt
import torch
print(torch.cuda.is_available())
torch.cuda.set_device(0) # 设置为第一个 GPU
logger = trt.Logger(trt.Logger.WARNING)
builder = trt.Builder(logger)
print('asfasd')